import os


class Class:

    def __init__(self, name_list_path, target_path):
        self.name_list_path = name_list_path
        self.target_path = target_path
        self.name_list = []
        self.target_list = []

    def read_name_list(self):
        with open(self.name_list_path, 'r') as f:
            data = f.readlines()
        for i in data:
            temp = i.split('\t')
            self.name_list.append([temp[0], temp[1][:-1]])
        return self.name_list

    def read_target_list(self):
        self.target_list = os.listdir(self.target_path)
        # self.target_list.remove('.DS_Store')
        return self.target_list
